In [ ]:
# Ruta a la carpeta con las imágenes de malware
dir_path = "malimg_paper_dataset_imgs"
min_obs = 100
for family in os.listdir(dir_path):
family_dir = os.path.join(dir_path, family)
if os.path.isdir(family_dir):
count = 0
for filename in os.listdir(family_dir):
if filename.endswith('.png'):
im = Image.open(os.path.join(family_dir, filename))
count += 1
if count >= min_obs:
fig, axs = plt.subplots(3, 3, figsize=(12,12))
fig.suptitle(family, fontsize=20)
i = 0
for filename in os.listdir(family_dir):
if filename.endswith('.png'):
im = Image.open(os.path.join(family_dir, filename))
axs[i//3, i%3].imshow(im)
axs[i//3, i%3].set_title(family)
i += 1
if i == 9:
break
plt.show()
# Muestra el número de imágenes por familia de malware
